# -!- coding: utf-8 -!-
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
#作者：cacho_37967865
#博客：https://blog.csdn.net/sinat_37967865
#文件：baidu_antiporn.py
#日期：2019-11-09
#备注：Python利用百度AI进行图像审核服务, 需要先安装python SDK   pip install baidu-aip
    https://cloud.baidu.com/doc/ANTIPORN/index.html
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''

from aip import AipImageCensor
from dealFile import get_file
from dealJson import dict_to_json

# 定义常量
APP_ID = '17732245'
API_KEY = '9mYSCNj1iFIvUsfU7vNfVjz9'
SECRET_KEY = 'GjCQdgC1FCs2eNVPdDHY1DGAF4hEeyvZ'

client = AipImageCensor(APP_ID, API_KEY, SECRET_KEY)


# 图像审核接口：为用户提供色情识别、暴恐识别、政治敏感人物识别、广告识别、图像垃圾文本识别（反作弊）、
# 恶心图像识别等一系列图像识别接口的一站式服务调用
def image_censor(file):
    image = get_file(file)
    result = client.imageCensorUserDefined(image)
    dict_to_json(result)


# GIF色情图像识别
def antiPorn_gif(file):
    image = get_file(file)
    result = client.antiPornGif(image)
    dict_to_json(result)


# 头像审核-》识别详细结果：ocr-文字识别，face-人脸识别，star-明星人脸，public-公众人物，
# politicians-政治人物，antiporn-色情识别，terror-暴恐识别
def face_audit(file):
    image = get_file(file)
    result = client.faceAudit(image)
    dict_to_json(result)


# 组合审核：ocr-通用文字识别，public-公众人物识别，politician-政治人物识别，antiporn-色情识别，terror-暴恐识别，
# webimage-网图OCR识别，disgust-恶心图，watermark-水印、二维码
def imageCensor_comb(file):
    image = get_file(file)
    result = client.imageCensorComb(image,['public','antiporn'])
    dict_to_json(result)


# 运用业界领先的深度学习技术，判断一段文本内容是否符合网络发文规范，实现自动化、智能化的文本审核。
# 审核内容目前分为5大方向：色情文本、政治敏感、恶意推广、网络暴恐、低俗辱骂。
def anti_spam():
    word = '垃圾文本'
    result = client.antiSpam(word)
    dict_to_json(result)


if __name__ == '__main__':
    #image_censor('ocr/lyf.jpg')
    #antiPorn_gif('images/PYG02-ball.gif')
    #face_audit('ocr/lyf.jpg')
    #imageCensor_comb('ocr/lyf.jpg')
    anti_spam()



